# 创建测试脚本 test_tensorrt.py
import tensorrt as trt

print("TensorRT 版本:", trt.__version__)

# 创建 logger
logger = trt.Logger(trt.Logger.WARNING)

# 尝试创建 builder
builder = trt.Builder(logger)

if builder:
    print("TensorRT 安装成功!")
else:
    print("TensorRT 安装失败!")